import os

import yaml

from class23test0515.page_object.basepage import BasePage


class LoginPage(BasePage):
    def login(self):
        from class23test0515.page_object.homepage import HomePage
        # 1,Basepage 的init方法 打开了企业微信，但是需要登录
        # 2,获取cookie，加到driver中
        print(os.path.abspath('..'))
        with open('../data/cookies.yaml', 'r') as f:
            cookies = yaml.safe_load(f)
        for cookie in cookies:
            self.driver.add_cookie(cookie)
        # 3,再次打开企业微信
        self.driver.get("https://work.weixin.qq.com/wework_admin/frame")
        return HomePage(self.driver)
